# Импорты
import pygame
import sys
import random
import time

#Инициализация PyGame
pygame.init()

screen = pygame.display.set_mode((500, 500)) #Окно с размерами 500x500
pygame.display.set_caption('Молния') #Название окна

screen.fill((0, 0, 0)) #Заливка экрана

line = pygame.Surface((2, 1)) #Создание поверхности
line.fill((250, 255, 0)) #Заливка поверхности

#Создание переменных
y = i = x = 0
y_size = 1
lightning = False

#События
def ev():
	global x, lightning, y_size #Глобальные переменные

	for event in pygame.event.get(): #Цикл событий

		if event.type == pygame.QUIT: #Закрытие программы при клике на крестик
			pygame.quit() #Выход из PyGame
			sys.exit() #Выход из программы

		elif event.type == pygame.MOUSEBUTTONDOWN and lightning == False: #Событие клик мышкой
			if event.button == 1: #Левый клик
				x, y_size = pygame.mouse.get_pos() #Получение координат мышки
				lightning = True # Переменная для молнии True

while True: #Главный цикл


	for value in range(y_size):
		if lightning: # Цикл генерации молнии
			y += 1
			i += 1
			x += random.randint(-1, 1)
				
			screen.blit(line, (x, y)) #Отображение молнии

		ev() #Вызов функции с событиями
		pygame.display.update() #Обновление экрана

		if i >= y_size: #Сброс молнии
			lightning = False
			i = 0
			y = 0
			screen.fill((0, 0, 0))

